package com.xigege.netty.protocolDesignAndAnalysis.customProtocol.protocol;


import com.xigege.netty.protocolDesignAndAnalysis.customProtocol.message.Message;
import com.xigege.netty.protocolDesignAndAnalysis.customProtocol.protocol.config.Config;
import io.netty.buffer.ByteBuf;
import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.MessageToMessageCodec;

import java.util.List;

/**
 * 实现的功能和 MessageCodec 一样
 * 由于 MessageCodec 父类规定，不可加上@Sharable注解，不可共享
 * （实际上：经由LengthFieldBasedFrameDecoder解码器，MessageCodec中的信息一定是完整的，故可被共享）
 * <p>
 * MessageCodecSharable 是可被共享的（父类也没限制），可以节省一定的内存
 * 必须和LengthFieldBasedFrameDecoder一起使用，确保接受到的ByteBuf消息是完整的
 */
@ChannelHandler.Sharable
public class MessageCodecSharable extends MessageToMessageCodec<ByteBuf, Message> {
    @Override
    protected void encode(ChannelHandlerContext ctx, Message msg, List<Object> outList) throws Exception {
        ByteBuf out = ctx.alloc().buffer();
        // 1.写入魔数 四个字节
        out.writeBytes(new byte[]{1, 0, 1, 4});
        // 2.版本号 一个字节
        out.writeByte(1);
        // 3.序列化算法 一个字节
        // 枚举顺序：0代表jdk序列方式，1代表json序列化方式
        out.writeByte(Config.getSerializerAlgorithm().ordinal());
        // 4.指令类型 一个字节
        out.writeByte(msg.getMessageType());
        // 5.请求序号 四个字节
        out.writeInt(msg.getSequenceId());
        //写入一个字节的对其填充，使除消息正文的消息的字节数为2的整数倍（16）
        out.writeByte(0xff);
        // 序列化msg
        byte[] bytes = Config.getSerializerAlgorithm().serialize(msg);
        // 6.正文长度 四个字节
        out.writeInt(bytes.length);
        // 7.消息正文
        out.writeBytes(bytes);

        outList.add(out);
    }

    @Override
    protected void decode(ChannelHandlerContext ctx, ByteBuf in, List<Object> out) throws Exception {
        // 读取四个字节的魔数
        int magicNum = in.readInt();
        // 读取一个字节的版本号
        byte version = in.readByte();
        // 读取一个字节的序列化算法
        byte serializeNum = in.readByte();
        // 读取一个字节的消息类型
        byte messageType = in.readByte();
        // 读取四个字节的请求序号
        int sequenceId = in.readInt();
        // 读取一个字节的对其填充
        byte padding = in.readByte();
        // 读取四个字节的正文长度
        int msgLength = in.readInt();
        // 读取消息正文
        byte[] msgBytes = new byte[msgLength];
        in.readBytes(msgBytes, 0, msgLength);

        // 反序列化msg
        // 1.找到反序列话算法
        Serializer.Algorithm algorithm = Serializer.Algorithm.values()[serializeNum];
        // 2. 根据messageType 找到真正的消息类型
        Class<? extends Message> messageClass = Message.getMessageClass(messageType);
        // 3.进行反序列化
        Message message = algorithm.deserialize(messageClass, msgBytes);

       // 添加到入站
        out.add(message);
        // 打印日志
        System.out.println("魔数:" + magicNum + "\n"
                + "version:" + version + "\n"
                + "serializeNum:" + serializeNum + "\n"
                + "messageType:" + messageType + "\n"
                + "sequenceId:" + sequenceId + "\n"
                + "padding:" + padding + "\n"
                + "msgLength:" + msgLength + "\n"
                + "message:" + message);
    }
}
